python list comprehension multiple ifs